var websocket = null;

//判断当前浏览器是否支持WebSocket, 主要此处要更换为自己的地址
if ('WebSocket' in window) {
    // websocket = new WebSocket("ws://localhost:18092/test/chatroom");
} else {
    alert('Not support websocket')
}

function initWebsocket(currentUser) {
    websocket = new WebSocket(`ws://localhost:18092/test/chatroom/${currentUser}`);

    //连接成功建立的回调方法
    websocket.onopen = function (event) {
        console.log("onopen")
    }

    //接收到消息的回调方法
    websocket.onmessage = function (event) {
        let text = `receive >> ${event.data}`
        let p = `<p style="background: aquamarine">${text}</p>`
        // 显示接收的信息
        setMessageInnerHTML(p);
    }

    //连接关闭的回调方法
    websocket.onclose = function () {
        console.log("onclose")
    }

    //连接发生错误的回调方法
    websocket.onerror = function () {
        console.error("error");
    };

    //监听窗口关闭事件，当窗口关闭时，主动去关闭websocket连接，防止连接还没断开就关闭窗口，server端会抛异常。
    window.onbeforeunload = function () {
        websocket.close();
    }
}

//将消息显示在网页上
function setMessageInnerHTML(innerHTML) {
    document.getElementById('message').innerHTML += innerHTML;

}

//关闭连接
function closeWebSocket() {
    websocket.close();
}

//发送消息
function send(data) {
    var message = document.getElementById('msg').value
    data.msg = message
    let text = `send >> ${message}`
    let p = `<p style="background: coral">${text}</p>`
    // 显示发送的信息
    setMessageInnerHTML(p);

    console.log(data)
    websocket.send(JSON.stringify(data));
}